import yaml


class YamlLoader:
    def __init__(self, config_dict='./src/config/config.yaml'):
        self.config_dict = config_dict
        self.config = None

    def get_config(self):
        # 读取 YAML 文件
        with open(self.config_dict, 'r') as file:
            self.config = yaml.safe_load(file)
        return self.config
